#pragma once

#include <iostream>
#include <string>
#include <queue>
#include <vector>
#include <memory>
#include "Mutex.hpp"
#include "Log.hpp"
#include "Thread.hpp"
#include "Cond.hpp"


namespace ThreadPoolModule
{
    using namespace MutexModule;
    using namespace LogModule;
    using namespace ThreadModule;
    using namespace CondModule;

    // 用来做测试的线程方法
    void DefaultTest()
    {
        while (true)
        {
            LOG(LogLevel::DEBUG) << "我是一个测试方法";
            sleep(1);
        }
    }

    using thread_t = std::shared_ptr<Thread>;

    const static int defaultnum = 5; // 默认的进程池进程数量

    template <typename T>
    class ThreadPool
    {
        bool IsEmpty() { return _taskq.empty(); }

        void HandlerTask(std::string name)
        {
            LOG(LogLevel::INFO) << "线程" << name << "进入HandlerTask逻辑";
            while (true)
            {
                // 1.拿任务
                T t;
                {
                    LockGuard lock(_mutex);
                    // 如果进程池没有退出，在任务列表为空时，要进入条件变量等待
                    while (IsEmpty() && isrunning)
                    {
                        _wait_num++;
                        _cond.Wait(_mutex);
                        _wait_num--;
                    }

                    // 如果进程池为退出状态，且任务列表为空，进程退出循环
                    if (IsEmpty() && !isrunning)
                        break;

                    t = _taskq.front();
                    _taskq.pop();
                }

                // 2.处理任务，不处于临界区，不需要拿着锁
                t(name);
            }
            LOG(LogLevel::INFO) << "线程: " << name << " 退出";
        }

        // 单例：只能有一个实例对象，不可随意创建新对象，构造函数不对外
        ThreadPool(int num = defaultnum) : _num(num), _wait_num(0), isrunning(false)
        {
            for (int i = 0; i < num; i++)
            {
                _threads.push_back(std::make_shared<Thread>(std::bind(&ThreadPool::HandlerTask, this, std::placeholders::_1)));
                LOG(LogLevel::INFO) << "构建线程" << _threads.back()->Name() << "对象 ... 成功";
            }
        }

    public:
        static ThreadPool<T> *getInstance()
        {
            // 如果多个线程同时进入，可能会有线程安全问题
            // 假设两个线程都要创建实例，刚开始都是null，都进了第一次开始争夺锁
            // 争抢到锁的那个进入第二个if，创建出实例
            // 第二个线程抢到锁后，不符合第二个if条件，不会再建立新的实例
            if (instance == NULL)
            {
                LockGuard lockguard(mutex);
                if (instance == NULL)
                {
                    LOG(LogLevel::INFO) << "单例首次被执行，需要加载对象...";
                    instance = new ThreadPool<T>();
                }
            }

            return instance;
        }

        void Equeue(T &&in)
        {
            // 3.如果进程退出，不再执行push任务的操作
            if (!isrunning)
                return;
            LockGuard lockguard(_mutex);
            _taskq.push(std::move(in));
            if (_wait_num)
                _cond.Notify();
        }

        void Start()
        {
            if (isrunning)
                return;

            isrunning = true;
            for (auto &thread_ptr : _threads)
            {
                thread_ptr->Start();
                LOG(LogLevel::INFO) << "启动线程" << thread_ptr->Name() << " ... 成功";
            }
        }

        void Wait()
        {
            for (auto &thread_ptr : _threads)
            {
                thread_ptr->Join();
                LOG(LogLevel::INFO) << "回收线程" << thread_ptr->Name() << " ... 成功";
            }
        }

        void Stop()
        {
            LockGuard lockgurad(_mutex);
            if (isrunning) // isrunning属于临界资源，只需要更改一次即可
            {
                isrunning = false; // 不工作

                // 进程退出：1.进程要自己退出（唤醒）
                // 2.任务列表的任务要执行完
                if (_wait_num)
                    _cond.NotifyAll();
            }
        }

    private:
        std::queue<T> _taskq;           // 任务列表：临界资源
        std::vector<thread_t> _threads; // 保存线程的指针，防止拷贝
        int _num;                       // 线程的数量
        int _wait_num;                  // 等待的线程数目

        Mutex _mutex;
        Cond _cond;
        bool isrunning;

        static ThreadPool<T> *instance;
        static Mutex mutex; // 只用来保护单例
    };

    // 静态变量的初始化，需要指定作用域
    template <typename T>
    ThreadPool<T> *ThreadPool<T>::instance = NULL;
    template <typename T>
    Mutex ThreadPool<T>::mutex; // 只用来保护单例

}